package leetcode.code1797;

import java.util.HashMap;
import java.util.Map;

public class AuthenticationManager {

	private Map<String, Integer> hash;

	private int timeToLive;

	public AuthenticationManager(int timeToLive) {
		hash = new HashMap<>();
		this.timeToLive = timeToLive;
	}

	public void generate(String tokenId, int currentTime) {
		hash.put(tokenId, currentTime + this.timeToLive);
	}

	public void renew(String tokenId, int currentTime) {
		if (hash.containsKey(tokenId)) {
			if (currentTime < hash.get(tokenId)) {
				hash.put(tokenId, this.timeToLive + currentTime);
			}
		}
	}

	public int countUnexpiredTokens(int currentTime) {
		int ans = 0;
		for (String tokenId : hash.keySet()) {
			if (hash.get(tokenId) > currentTime) {
				ans++;
			}
		}
		return ans;
	}

	public static void main(String[] args) {
		AuthenticationManager authenticationManager = new AuthenticationManager(5); // 构造 AuthenticationManager ，设置
																					// timeToLive = 5 秒。
		authenticationManager.renew("aaa", 1); // 时刻 1 时，没有验证码的 tokenId 为 "aaa" ，没有验证码被更新。
		authenticationManager.generate("aaa", 2); // 时刻 2 时，生成一个 tokenId 为 "aaa" 的新验证码。
		authenticationManager.countUnexpiredTokens(6); // 时刻 6 时，只有 tokenId 为 "aaa" 的验证码未过期，所以返回 1 。
		authenticationManager.generate("bbb", 7); // 时刻 7 时，生成一个 tokenId 为 "bbb" 的新验证码。
		authenticationManager.renew("aaa", 8); // tokenId 为 "aaa" 的验证码在时刻 7 过期，且 8 >= 7 ，所以时刻 8 的renew 操作被忽略，没有验证码被更新。
		authenticationManager.renew("bbb", 10); // tokenId 为 "bbb" 的验证码在时刻 10 没有过期，所以 renew 操作会执行，该 token 将在时刻 15 过期。
		authenticationManager.countUnexpiredTokens(15); // tokenId 为 "bbb" 的验证码在时刻 15 过期，tokenId 为 "aaa" 的验证码在时刻 7
														// 过期，所有验证码均已过期，所以返回 0 。
	}

}
